#!/usr/bin/env python
"""
创建基础学院数据的脚本
"""
import os
import django

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'educational_system.settings')
django.setup()

from apps.accounts.models import College

def create_colleges():
    """创建基础学院数据"""
    colleges_data = [
        {'name': '计算机科学与技术学院', 'code': 'CS', 'description': '计算机科学与技术相关专业'},
        {'name': '电子信息工程学院', 'code': 'EE', 'description': '电子信息工程相关专业'},
        {'name': '机械工程学院', 'code': 'ME', 'description': '机械工程相关专业'},
        {'name': '经济管理学院', 'code': 'EM', 'description': '经济管理相关专业'},
        {'name': '外国语学院', 'code': 'FL', 'description': '外国语相关专业'},
        {'name': '艺术设计学院', 'code': 'AD', 'description': '艺术设计相关专业'},
        {'name': '数学与统计学院', 'code': 'MS', 'description': '数学与统计相关专业'},
        {'name': '物理与材料学院', 'code': 'PM', 'description': '物理与材料相关专业'},
    ]
    
    created_count = 0
    for college_data in colleges_data:
        college, created = College.objects.get_or_create(
            name=college_data['name'],
            defaults={
                'code': college_data['code'],
                'description': college_data['description']
            }
        )
        if created:
            created_count += 1
            print(f"创建学院: {college.name}")
        else:
            print(f"学院已存在: {college.name}")
    
    print(f"\n总共创建了 {created_count} 个学院")
    print(f"当前共有 {College.objects.count()} 个学院")

if __name__ == '__main__':
    create_colleges()